GIE_Context Class Reference

Detailed Description

Definition at line 126 of file gie_inference.h.

Collaboration diagram for GIE_Context:

Public Member Functions
void	gie_alloc ()
void	gie_free ()
void	caffeToGIEModel ()
void	doInference (float *input, queue< vector< cv::Rect > > &rectList_queue)
int	parse_net ()

Data Fields
unsigned int	net_width
unsigned int	net_height
unsigned int	image_width
unsigned int	image_height
unsigned int	filter_num
void *	buffers [NUM_BINDINGS]
float *	input_buf
float *	output_cov_buf
float *	output_bbox_buf
float	helnet_scale [4]
int	hel_net
IRuntime *	runtime
ICudaEngine *	engine
IExecutionContext *	context
uint32_t *	pResultArray
string	deployfile
string	modelfile
unsigned int	channel
unsigned int	num_bindings
unsigned int	inference_num
unsigned int	batch_size
stringstream	gieModelStream
vector< string >	outputs
class Logger	Logger
struct Profiler	Profiler
unsigned int	forced_fp32
unsigned int	dump_result
string	result_file
ofstream	fstream
uint32_t	enable_gie_profiler

Member Function Documentation

void GIE_Context::caffeToGIEModel

(

)

void GIE_Context::doInference	(	float *	input,
		queue< vector< cv::Rect > > &	rectList_queue
	)

void GIE_Context::gie_alloc

(

)

void GIE_Context::gie_free

(

)

int GIE_Context::parse_net

(

)

Field Documentation

unsigned int GIE_Context::batch_size

Definition at line 149 of file gie_inference.h.

void* GIE_Context::buffers[NUM_BINDINGS]

Definition at line 134 of file gie_inference.h.

unsigned int GIE_Context::channel

Definition at line 146 of file gie_inference.h.

IExecutionContext* GIE_Context::context

Definition at line 142 of file gie_inference.h.

string GIE_Context::deployfile

Definition at line 144 of file gie_inference.h.

unsigned int GIE_Context::dump_result

Definition at line 155 of file gie_inference.h.

uint32_t GIE_Context::enable_gie_profiler

Definition at line 158 of file gie_inference.h.

ICudaEngine* GIE_Context::engine

Definition at line 141 of file gie_inference.h.

unsigned int GIE_Context::filter_num

Definition at line 133 of file gie_inference.h.

unsigned int GIE_Context::forced_fp32

Definition at line 154 of file gie_inference.h.

ofstream GIE_Context::fstream

Definition at line 157 of file gie_inference.h.

stringstream GIE_Context::gieModelStream

Definition at line 150 of file gie_inference.h.

int GIE_Context::hel_net

Definition at line 139 of file gie_inference.h.

float GIE_Context::helnet_scale[4]

Definition at line 138 of file gie_inference.h.

unsigned int GIE_Context::image_height

Definition at line 132 of file gie_inference.h.

unsigned int GIE_Context::image_width

Definition at line 131 of file gie_inference.h.

unsigned int GIE_Context::inference_num

Definition at line 148 of file gie_inference.h.

float* GIE_Context::input_buf

Definition at line 135 of file gie_inference.h.

class Logger GIE_Context::Logger

Definition at line 152 of file gie_inference.h.

string GIE_Context::modelfile

Definition at line 145 of file gie_inference.h.

unsigned int GIE_Context::net_height

Definition at line 130 of file gie_inference.h.

unsigned int GIE_Context::net_width

Definition at line 129 of file gie_inference.h.

unsigned int GIE_Context::num_bindings

Definition at line 147 of file gie_inference.h.

float* GIE_Context::output_bbox_buf

Definition at line 137 of file gie_inference.h.

float* GIE_Context::output_cov_buf

Definition at line 136 of file gie_inference.h.

vector<string> GIE_Context::outputs

Definition at line 151 of file gie_inference.h.

uint32_t* GIE_Context::pResultArray

Definition at line 143 of file gie_inference.h.

struct Profiler GIE_Context::Profiler

Definition at line 153 of file gie_inference.h.

string GIE_Context::result_file

Definition at line 156 of file gie_inference.h.

IRuntime* GIE_Context::runtime

Definition at line 140 of file gie_inference.h.

---

The documentation for this class was generated from the following file:

• gie_inference.h